package org.example.me.handler.context;

import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import lombok.Data;

import java.util.HashMap;
import java.util.Map;

/**
 * 文件上下文
 */
@Data
public class FileContext {

    /**
     * fileName 标识，同个 fileName 只能生成一个文件
     */
    private String fileName;

    /**
     * 一个文件只能用一个 写入器
     * 不然一直会覆盖第一个
     */
    private ExcelWriter excelWriter;

    /**
     * sheet名称和上下文
     */
    private Map<String, WriteSheet> writeSheetMap;

    /**
     * 添加Sheet到文件上下文中
     * @param writeSheet Sheet上下文
     */
    public void addWriteSheet(WriteSheet writeSheet) {
        if (this.writeSheetMap == null) {
            this.writeSheetMap = new HashMap<>();
        }
        writeSheetMap.put(writeSheet.getSheetName(), writeSheet);
    }

}
